package com.atguigu.tingshu.album.service.impl;

import com.alibaba.fastjson.JSONObject;
import com.atguigu.tingshu.album.mapper.*;
import com.atguigu.tingshu.album.service.AlbumInfoService;
import com.atguigu.tingshu.album.service.BaseCategoryService;
import com.atguigu.tingshu.model.album.*;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;

@Service
@SuppressWarnings({"all"})
public class BaseCategoryServiceImpl extends ServiceImpl<BaseCategory1Mapper, BaseCategory1> implements BaseCategoryService {

    @Autowired
    private BaseCategory1Mapper baseCategory1Mapper;

    @Autowired
    private BaseCategory2Mapper baseCategory2Mapper;

    @Autowired
    private BaseCategory3Mapper baseCategory3Mapper;


    @Autowired
    private BaseCategoryViewMapper baseCategoryViewMapper;

    @Autowired
    private BaseAttributeMapper baseAttributeMapper;

    //根据一级分类id查询对应标签数据（标签名称和标签值）
    @Override
    public List<BaseAttribute> findAttribute(Long category1Id) {
        //调用mapper
        return baseAttributeMapper.selectAttribute(category1Id);
    }

    /**
     * 通过三级分类id查询分类信息
     * @param category3Id
     * @return
     */
    @Override
    public BaseCategoryView getCategoryView(Long category3Id) {
        BaseCategoryView baseCategoryView = baseCategoryViewMapper.selectById(category3Id);
        return baseCategoryView;
    }

    /**
     * 根据一级分类Id 查询置顶频道页的三级分类列表
     * @param category1Id
     * @return
     */
    @Override
    public List<BaseCategory3> selectTopBaseCategory3(Long category1Id) {
        //    # 第一步 根据一级分类id查询下面所有二级分类id
        //    SELECT bc.id FROM base_category2 bc WHERE bc.category1_id=1;
        //    # 101  102  103
        LambdaQueryWrapper<BaseCategory2> wrapper2 = new LambdaQueryWrapper<>();
        wrapper2.eq(BaseCategory2::getCategory1Id, category1Id);
        List<BaseCategory2> baseCategory2List = baseCategory2Mapper.selectList(wrapper2);

        // baseCategory2List二级分类所有数据集合，获取所有二级分类id，构建新集合
        List<Long> category2IdList = baseCategory2List.stream()
                .map(BaseCategory2::getId)
                .collect(Collectors.toList());

        //    # 第二步 根据第一步查询出来所有二级分类id，查询下面三级分类数据，
        //    #         有条件是否置顶 is_top=1,
        //    SELECT * FROM base_category3 bc3 WHERE bc3.category2_id IN(101,102,103)
        //    AND  bc3.is_top=1  LIMIT 7
        LambdaQueryWrapper<BaseCategory3> wrapper3 = new LambdaQueryWrapper<>();
        wrapper3.in(BaseCategory3::getCategory2Id, category2IdList);
        wrapper3.eq(BaseCategory3::getIsTop, 1);
        wrapper3.last(" limit 7");
        List<BaseCategory3> baseCategory3List = baseCategory3Mapper.selectList(wrapper3);
        return baseCategory3List;
    }

    /**
     * 根据一级分类Id 获取全部数据
     * @param category1Id
     * @return
     */
    @Override
    public JSONObject getAllCategoryList(Long category1Id) {
        //是用于点击三级分类栏的全部搜索框时，将一级分类对应的二级分类和三级分类信息都查询出来,分别填充到对应位置
        //1、先根据一级分类ID获取到一级分类的信息
        BaseCategory1 baseCategory1 = baseCategory1Mapper.selectById(category1Id);
        // 声明一级分类对象
        JSONObject category1 = new JSONObject();
        category1.put("categoryId", category1Id);
        category1.put("categoryName", baseCategory1.getName());

        //通过视图查询获取信息里面的二级和三级信息
        LambdaQueryWrapper<BaseCategoryView> wrapper = new LambdaQueryWrapper<>();
        wrapper.eq(BaseCategoryView::getCategory1Id, category1Id);
        List<BaseCategoryView> baseCategoryViewlist = baseCategoryViewMapper.selectList(wrapper);

        //根据二级分类id分组，转换数据 key是二级分类id，value是对应的值
        Map<Long, List<BaseCategoryView>> map = baseCategoryViewlist.stream()
                .collect(Collectors.groupingBy(BaseCategoryView::getCategory2Id));

        //遍历map集合，封装二级分类数据，最终把二级集合放到一级分类里面
        //创建一个准备放到一级分类里面的list集合对象
        List<JSONObject> category2Child = new ArrayList<>();

        map.forEach((k, v) -> {
            Long categoryId2 = k;
            List<BaseCategoryView> baseCategoryViewList2 = v;

            //封装二级分类数据
            JSONObject category2 = new JSONObject();
            category2.put("categoryId", categoryId2);
            category2.put("categoryName", baseCategoryViewList2.get(0).getCategory2Name());

            // 封装三级，把三级放到二级分类里面
            // List<BaseCategoryView> -- List<JSONObject>
            List<JSONObject> category3Child = new ArrayList<>();
            baseCategoryViewList2.stream().forEach(category3View -> {
                JSONObject category3 = new JSONObject();
                category3.put("categoryId", category3View.getCategory3Id());
                category3.put("categoryName", category3View.getCategory3Name());
                category3Child.add(category3);
            });
            //把三级数据集合放到二级里面
            category2.put("categoryChild",category3Child);

            //多个二级分类放到集合里面
            category2Child.add(category2);
        });

        //将二级分类添加到一级分类里面
        category1.put("categoryChild",category2Child);
        return category1;
    }

    /**
     * 查询所有的分类数据
     * @return //三级分类，思路：观察最后需要返回的json格式的数据，我们发现，整个是一个大的集合，里面每一个元素是一个一级分类
     * //每个一级分类（map）里面的categorychild字段对应着一个集合，里面每一个元素都是二级分类（map）
     * //每个二级分类（map）里面的categorychild字段对应着一个集合，里面每一个元素都是三级分类
     * //步骤：
     * //1、创建一个集合，对一级分类进行分组得到一个map，遍历这个map,因为这里确保每个一级至少有一个二级，每个二级至少有一个三级
     * //获取里面的第一行数据的id和名称来作为一级分类的id和name
     * //2、创建一个集合，对二级分类进行分组得到一个map，遍历这个map，获取里面的第一行数据的id和名称来作为二级分类的id和name
     * //3、创建一个集合，存放三级分类，因为三级分类存在二级分类里面，只是类型不同，所以我们直接遍历二级分类，然后创建一个map，遍历，直接获取三级分类的id和name，添加到map中即可
     * //最后，分别将对应的集合和categorychild字段对应
     * //返回一个最终的集合
     */
    @Override
    public List<JSONObject> getBaseCategoryList() {


        //最终的集合
        List<JSONObject> finalList = new ArrayList<>();

        //1、我们首先需要获取数据，把视图里面的分类都拿出来,三级分类的数据都存在视图中，方便之后的查询
        List<BaseCategoryView> baseCategoryViewList =
                baseCategoryViewMapper.selectList(null);
        //根据一级分类id进行分组
        Map<Long, List<BaseCategoryView>> map1 = baseCategoryViewList.stream()
                .collect(Collectors.groupingBy(BaseCategoryView::getCategory1Id));
        //遍历一级分类
        // map的key：分组字段，一级分类id
        // map的value：每组里面所有数据集合
        map1.forEach((k1, v1) -> {
            Long categoryId1 = k1; //一级分类id
            List<BaseCategoryView> baseCategoryViewList1 = v1; //每组一级分类组所有数据集合
            //封装一级分类，为了之后添加到总的集合中
            JSONObject jsonObject1 = new JSONObject();
            //key value的形式，因为至少有1个一级分类，所以从0号开始是最安全的，数量不确定，如果设置其他数字，可能空指针
            jsonObject1.put("categoryId", categoryId1);
            jsonObject1.put("categoryName", baseCategoryViewList1.get(0).getCategory1Name());


            //根据二级分类进行分组,针对的是一级分类的子集进行再分类
            Map<Long, List<BaseCategoryView>> map2 = baseCategoryViewList1.stream()
                    .collect(Collectors.groupingBy(BaseCategoryView::getCategory2Id));

            //创建一个存放二级分类的集合,用于放在一级集合的child字段处
            List<JSONObject> categoryChild2 = new ArrayList<>();

            //遍历整个分组map
            map2.forEach((k2, v2) -> {
                Long categoryId2 = k2; //二级分类id
                List<BaseCategoryView> baseCategoryViewList2 = v2; //每组二级分类组所有数据集合

                //封装二级分类，获取每个二级分类的id和name
                JSONObject jsonObject2 = new JSONObject();
                jsonObject2.put("categoryId", categoryId2);
                jsonObject2.put("categoryName", baseCategoryViewList2.get(0).getCategory2Name());

                //获取三级分类的id和name,因为三级分类没有子分类了，三级分类的内容在二级分类分组之后的内容中有
                //所以他们可以通过二级分类的map来获得
                //这个的类型是一个list,作为二级分类的categoryChild字段，我们通过流的方式来取代循环,获取三级分类
                List<JSONObject> categoryChild3 = baseCategoryViewList2.stream().map(baseCategoryView -> {
                    //封装三级分类
                    JSONObject jsonObject3 = new JSONObject();
                    jsonObject3.put("categoryId", baseCategoryView.getCategory3Id());
                    jsonObject3.put("categoryName", baseCategoryView.getCategory3Name());
                    return jsonObject3;
                }).collect(Collectors.toList());

                //把三级集合放到每个二级分类对象里面
                jsonObject2.put("categoryChild", categoryChild3);

                //将二级分类的对象要添加到集合中
                categoryChild2.add(jsonObject2);
            });

            //添加一级分类的categoryChild字段
            jsonObject1.put("categoryChild", categoryChild2);

            //将一级分类添加到最终的集合中
            finalList.add(jsonObject1);
        });

        return finalList;
    }
}
